import csv
from faker import Faker

fake = Faker()

# 创建一个集合来存储已经生成的用户名
names = set()

# 生成数据
data = []
for i in range(500):
    username = fake.user_name().replace(' ', '')
    # 检查用户名是否已经存在
    while username in names:
        # 如果已经存在，则重新生成一个新的用户名
        username = fake.name().replace(' ', '')
    # 将新生成的用户名添加到集合中
    names.add(username)
    password = fake.password(special_chars=False, digits=True, upper_case=True, lower_case=True)
    data.append([username, password, '100'])


# 写入CSV文件
with open('data_user.csv', 'w', newline='', encoding='utf-8') as file:
    writer = csv.writer(file)
    writer.writerows(data)

print('数据已保存到data_user.csv文件中')
